import pandas as pd
import matplotlib.pyplot as plt
from matplotlib import rcParams
# 解决中文乱码
rcParams['font.sans-serif'] = ['SimHei']  # 使用黑体 SimHei
rcParams['axes.unicode_minus'] = False   
# 定义路径
output_file = r"F:\network-collect\compare\对比结果.xlsx"

# 从 Excel 读取不同工作表的数据
result_da_xi = pd.read_excel(output_file, sheet_name="大连-西安")
result_xi_da = pd.read_excel(output_file, sheet_name="西安-大连")

# 可视化函数
def visualize_comparison(result_df, direction):
    # 对 "票价对比" 列进行统计
    comparison_counts = result_df["票价对比"].value_counts()

    # 创建柱状图
    plt.figure(figsize=(8, 6))
    comparison_counts.plot(kind="bar", color=["skyblue", "lightgreen", "orange"])
    plt.title(f"{direction} 票价对比统计", fontsize=14)
    plt.xlabel("票价变化", fontsize=12)
    plt.ylabel("数量", fontsize=12)
    plt.xticks(rotation=0, fontsize=10)
    plt.grid(axis="y", linestyle="--", alpha=0.7)
    plt.tight_layout()

    # 展示图像
    plt.show()

# 分别对“大连-西安”和“西安-大连”的数据进行可视化
visualize_comparison(result_da_xi, "大连-西安")
visualize_comparison(result_xi_da, "西安-大连")
